# bash completion for aspell                               -*- shell-script -*-

_aspell_dictionary()
{
    local datadir aspell=${1:-aspell}
    datadir=$($aspell config data-dir 2>/dev/null || echo /usr/lib/aspell)
    # First, get aliases (dicts dump does not list them)
    COMPREPLY=($datadir/*.alias)
    if ((${#COMPREPLY[@]})); then
        COMPREPLY=("${COMPREPLY[@]%.alias}")
        COMPREPLY=("${COMPREPLY[@]#$datadir/}")
    fi
    # Then, add the canonical dicts
    COMPREPLY+=($($aspell dicts 2>/dev/null))
    ((${#COMPREPLY[@]})) &&
        COMPREPLY=($(compgen -X '\*' -W '"${COMPREPLY[@]}"' -- "$cur"))
}

_aspell()
{
    local cur prev words cword split
    _init_completion -s || return

    case $prev in
        -c | -p | check | --conf | --personal | --repl | --per-conf)
            _filedir
            return
            ;;
        --conf-dir | --data-dir | --dict-dir | --home-dir | --local-data-dir | --prefix)
            _filedir -d
            return
            ;;
        dump | create | merge)
            COMPREPLY=($(compgen -W 'master personal repl' -- "$cur"))
            return
            ;;
        --mode)
            COMPREPLY=($(compgen -W "$($1 modes 2>/dev/null |
                awk '{ print $1 }')" -- "$cur"))
            return
            ;;
        --sug-mode)
            COMPREPLY=($(compgen -W 'ultra fast normal bad-speller' \
                -- "$cur"))
            return
            ;;
        --keymapping)
            COMPREPLY=($(compgen -W 'aspell ispell' -- "$cur"))
            return
            ;;
        -d | --master)
            _aspell_dictionary "$1"
            return
            ;;
        --add-filter | --rem-filter)
            COMPREPLY=($(compgen -W "$($1 filters 2>/dev/null |
                awk '{ print $1 }')" -- "$cur"))
            return
            ;;
    esac

    $split && return

    if [[ $cur == -* ]]; then
        COMPREPLY=($(compgen -W '--conf= --conf-dir= --data-dir= --dict-dir=
            --encoding= --add-filter= --rem-filter= --mode= --add-extra-dicts=
            --rem-extra-dicts= --home-dir= --ignore= --ignore-accents
            --dont-ignore-accents --ignore-case --dont-ignore-case
            --ignore-repl --dont-ignore-repl --jargon --keyboard= --lang=
            --language-tag --local-data-dir= --master= --module
            --add-module-search-order --rem-module-search-order --per-conf=
            --personal= --prefix= --repl= --run-together --dont-run-together
            --run-together-limit= --run-together-min= --save-repl
            --dont-save-repl --set-prefix --dont-set-prefix --size= --spelling
            --strip-accents --dont-strip-accents --sug-mode=
            --add-word-list-path --rem-word-list-path --backup --dont-backup
            --reverse --dont-reverse --time --dont-time --keymapping=
            --add-email-quote= --rem-email-quote= --email-margin=
            --add-tex-command= --rem-tex-command= --tex-check-comments
            --dont-tex-check-comments --add-tex-extension --rem-tex-extension
            --add-sgml-check= --rem-sgml-check= --add-sgml-extension
            --rem-sgml-extension' -- "$cur"))
        [[ ${COMPREPLY-} == *= ]] && compopt -o nospace
    else
        COMPREPLY=($(compgen -W 'usage help check pipe list config soundslike
            filter version dump create merge' -- "$cur"))
    fi
} &&
    complete -F _aspell aspell

# ex: filetype=sh
